/*
 * Copyright 2016 Red Hat, Inc. and/or its affiliates
 * and other contributors as indicated by the @author tags.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.keycloak.social.dingtalk;

import org.keycloak.broker.oidc.mappers.AbstractJsonUserAttributeMapper;
import org.keycloak.broker.provider.BrokeredIdentityContext;
import org.keycloak.models.IdentityProviderMapperModel;
import org.keycloak.models.KeycloakSession;
import org.keycloak.models.RealmModel;
import org.keycloak.models.UserModel;

/**
 * User attribute mapper.
 * 
 * @author Vlastimil Elias (velias at redhat dot com)
 */
public class DingtalkUserAttributeMapper extends AbstractJsonUserAttributeMapper {

	public static final String PROVIDER_ID = "dingtalk-user-attribute-mapper";
	private static final String[] cp = new String[] { DingtalkIdentityProviderFactory.PROVIDER_ID };

	@Override
	public String[] getCompatibleProviders() {
		return cp;
	}

	@Override
	public String getId() {
		return PROVIDER_ID;
	}

	
    @Override
    public void updateBrokeredUser(KeycloakSession session, RealmModel realm, UserModel user, IdentityProviderMapperModel mapperModel, BrokeredIdentityContext context) {
        
        user.setSingleAttribute(DingtalkIdentityProvider.PROFILE_IDP_TYPE, context.getUserAttribute(DingtalkIdentityProvider.PROFILE_IDP_TYPE));
        user.setSingleAttribute(DingtalkIdentityProvider.PROFILE_NICK, context.getUserAttribute(DingtalkIdentityProvider.PROFILE_NICK));
        user.setSingleAttribute(DingtalkIdentityProvider.PROFILE_UNIONID, context.getUserAttribute(DingtalkIdentityProvider.PROFILE_UNIONID));
        user.setSingleAttribute(DingtalkIdentityProvider.PROFILE_OPENID, context.getUserAttribute(DingtalkIdentityProvider.PROFILE_OPENID));
        //拓展标记
        user.setSingleAttribute(DingtalkIdentityProvider.ATTRIBUTE_PREFIX+DingtalkIdentityProvider.PROFILE_NICK, context.getUserAttribute(DingtalkIdentityProvider.PROFILE_NICK));
        user.setSingleAttribute(DingtalkIdentityProvider.ATTRIBUTE_PREFIX+DingtalkIdentityProvider.PROFILE_UNIONID, context.getUserAttribute(DingtalkIdentityProvider.PROFILE_UNIONID));
        
        user.setUsername(context.getUsername());
        user.setEmail(context.getEmail());
        user.setFirstName(context.getFirstName());
        user.setLastName(context.getLastName());
    }
	
}
